Fixed bug with missing code.