Fixed brakeman issue