Fixing stupid bug

Fixing stupid bug in initial commit, stupid should have tested..