Fixed imports and stupid syntax error