diff --git a/syntax/prisma.vim b/syntax/prisma.vim
index 57adbe2..a6a9286 100644
--- a/syntax/prisma.vim
+++ b/syntax/prisma.vim
@@ -5,7 +5,7 @@ endif
 syn case match
 
 " Comment
-syn match prismaComment "\v//.*$"
+syn match prismaComment "\v\s*//.*$"
 " Directive
 syn match prismaDirective /\<@@\=\h\w*/ nextgroup=prismaFunctionParans
 " Ugly hack right now, probably this is not needed after refactoring
@@ -24,7 +24,7 @@ syn match prismaType /\<\h\w*\>/ contained containedin=prismaModelDeclaration ne
 syn region prismaFieldRegion start=/\v^\s*/ms=e+1 end=/\v\s/me=s-1 contains=prismaField,prismaComment contained transparent containedin=prismaModelDeclaration skipwhite
 syn match prismaMultiFieldDirective /^\s*@/ contained containedin=prismaModelDeclaration nextgroup=prismaPartialDirective
 
-syn region prismaNonModelDeclaration matchgroup=prismaModel start=/\v((datasource)=(generator)=(enum)=)+\s+\h\w*\s*\{/ end=/}/ contains=prismaString,prismaList,prismaValueDeclarationRegion,prismaOperator transparent
+syn region prismaNonModelDeclaration matchgroup=prismaModel start=/\v((datasource)=(generator)=(enum)=)+\s+\h\w*\s*\{/ end=/}/ contains=prismaString,prismaList,prismaValueDeclarationRegion,prismaOperator,prismaComment transparent
 syn match prismaValue /\<\h\w*\>/ contained containedin=prismaValueDeclarationRegion,prismaTypeAliasDeclaration nextgroup=prismaOperator skipwhite
 syn region prismaValueDeclarationRegion start=/\v^\s*/ms=e+1 end=/\v\s*/me=s-1 contains=prismaValueDeclaration contained transparent containedin=prismaNonModelDeclaration skipwhite
 
@@ -33,7 +33,6 @@ syn match prismaOperator "\[\]" display
 syn match prismaOperator /\v\=/ display
 syn region prismaList matchgroup=prismaList start="\[" end="]" contains=ALLBUT,prismaDirective,prismaModelDeclaration,prismaNonModelDeclaration
 
-
 hi def link prismaList Delimiter
 hi def link prismaParans Delimiter
 hi def link prismaDirective PreProc