Added code to fix bug.