Restore functionality of Rake tasks