[DO NOT MERGE] Fixed LMS tests for bok_choy version 0.3.0

@jzoldak