fix for the last pr