Fix some rake stuff