Fixed #37 by setting a viewmodel.