Fix rake issue