Re-fixed bug due to silly mistake