Fixed build() arguments
Fixed tests to call build() properly.
Merge pull request #7 from Nopik/fix_build_args

Fix build args