Fixes Test error

Update issue 105
Fix test error
Re-format