Minor bug fix while testing.