Fix for pr