Minor problem fix