Fix rake setup