Fixed minor Bug which came with the last commit