Fix grammar to work without module keyword