Quick minor fix