Initial commit fix

Forgot few things, oops. :)