Fix test failures

Merged in the changes from @viveksjain and fixed any test failures that came up.
I don't remember. Using padding would probably work better.