One more minor fix