Added small fix that was forgotten in last commit